import os
from dotenv import load_dotenv

load_dotenv()
os.environ["HF_HUB_OFFLINE"] = "1"

class Settings:
    # 路径配置
    PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    DATA_PATH = os.getenv("DATA_PATH", "./data/raw_documents")
    VECTOR_DB_PATH = os.getenv("VECTOR_DB_PATH", os.path.join(PROJECT_ROOT, "database/faiss_db"))
    MODEL_CACHE_PATH = os.getenv("MODEL_CACHE_PATH", os.path.join(PROJECT_ROOT, "model_cache"))
    # 模型配置
    EMBEDDING_MODEL = "BAAI/bge-small-zh-v1.5"
    # EMBEDDING_MODEL = "GanymedeNil/text2vec-large-chinese"
    LLM_MODEL = "deepseek-chat"  # DeepSeek模型

    # API密钥
    DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")


settings = Settings()
